Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_database_robustness.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

Tff7b72import T7ee787os

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72import T7ee787unittest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787provider Tff7b72import Te6edf3DatabaseProvider

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787schema Tff7b72import Te6edf3DatabaseSchema

Tff7b72class T56d364TestDatabaseRobustnessTb4b4b4(Te6edf3unittestTff7b72.Td2a8ffTestCaseTb4b4b4)Tb4b4b4:

Tff7b72def Td2a8ffsetUpTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

Tff7b72selfTff7b72.Td2a8fftest_dir Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffdb_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Tff7b72selfTff7b72.Td2a8fftest_dirTb4b4b4, Ta5d6ff"Ta5d6fftest_meshchat.dbTa5d6ff"Tb4b4b4)

T8b949e# Ensure we start with a fresh provider instance

Tff7b72if Tffa657hasattrTb4b4b4(Te6edf3DatabaseProviderTb4b4b4, Ta5d6ff"Ta5d6ff_instanceTa5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3DatabaseProviderTff7b72.Td2a8ff_instance Tff7b72= Tff7b72None

Tff7b72selfTff7b72.Td2a8ffprovider Tff7b72= Te6edf3DatabaseProviderTff7b72.Td2a8ffget_instanceTb4b4b4(Tff7b72selfTff7b72.Td2a8ffdb_pathTb4b4b4)

Tff7b72selfTff7b72.Td2a8ffschema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Tff7b72selfTff7b72.Td2a8ffproviderTb4b4b4)

Tff7b72def Td2a8fftearDownTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffclose_allTb4b4b4(Tb4b4b4)

Tff7b72if Tffa657hasattrTb4b4b4(Te6edf3DatabaseProviderTb4b4b4, Ta5d6ff"Ta5d6ff_instanceTa5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3DatabaseProviderTff7b72.Td2a8ff_instance Tff7b72= Tff7b72None

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Tff7b72selfTff7b72.Td2a8fftest_dirTb4b4b4)

Tff7b72def Td2a8fftest_missing_column_healingTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

T8b949e# 1. Create a "legacy" table without the peer_hash column

Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"""

Ta5d6ff CREATE TABLE lxmf_messages (

Ta5d6ff id INTEGER PRIMARY KEY AUTOINCREMENT,

Ta5d6ff hash TEXT UNIQUE,

Ta5d6ff source_hash TEXT,

Ta5d6ff destination_hash TEXT

Ta5d6ff )

Ta5d6ff Ta5d6ff"""Tb4b4b4)

T8b949e# 2. Also need the config table so initialize doesn't fail on version check

Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"""

Ta5d6ff CREATE TABLE config (

Ta5d6ff id INTEGER PRIMARY KEY AUTOINCREMENT,

Ta5d6ff key TEXT UNIQUE,

Ta5d6ff value TEXT

Ta5d6ff )

Ta5d6ff Ta5d6ff"""Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO config (key, value) VALUES (?, ?)Ta5d6ff"Tb4b4b4,

Tb4b4b4(Ta5d6ff"Ta5d6ffdatabase_versionTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ff1Ta5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4)

T8b949e# 3. Attempt initialization.

T8b949e# Previously this would crash with OperationalError: no such column: peer_hash

Tff7b72tryTb4b4b4:

Tff7b72selfTff7b72.Td2a8ffschemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:

Tff7b72selfTff7b72.Td2a8fffailTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffInitialization failed with missing column: Tffd700{Te6edf3eTffd700}Ta5d6ff"Tb4b4b4)

T8b949e# 4. Verify the column was added

Te6edf3cursor Tff7b72= Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffPRAGMA table_info(lxmf_messages)Ta5d6ff"Tb4b4b4)

Te6edf3columns Tff7b72= Tb4b4b4[Te6edf3rowTb4b4b4[T79c0ff1Tb4b4b4] Tff7b72for Te6edf3row Tff7b72in Te6edf3cursorTff7b72.Td2a8fffetchallTb4b4b4(Tb4b4b4)Tb4b4b4]

Tff7b72selfTff7b72.Td2a8ffassertInTb4b4b4(Ta5d6ff"Ta5d6ffpeer_hashTa5d6ff"Tb4b4b4, Te6edf3columnsTb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertInTb4b4b4(Ta5d6ff"Ta5d6ffis_spamTa5d6ff"Tb4b4b4, Te6edf3columnsTb4b4b4)

Tff7b72def Td2a8fftest_corrupt_config_initializationTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

T8b949e# 1. Create a database where the version is missing or garbled

Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"""

Ta5d6ff CREATE TABLE config (

Ta5d6ff id INTEGER PRIMARY KEY AUTOINCREMENT,

Ta5d6ff key TEXT UNIQUE,

Ta5d6ff value TEXT

Ta5d6ff )

Ta5d6ff Ta5d6ff"""Tb4b4b4)

T8b949e# No version inserted

T8b949e# 2. Initialization should still work

Tff7b72tryTb4b4b4:

Tff7b72selfTff7b72.Td2a8ffschemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:

Tff7b72selfTff7b72.Td2a8fffailTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffInitialization failed with missing version: Tffd700{Te6edf3eTffd700}Ta5d6ff"Tb4b4b4)

T8b949e# 3. Version should now be set to LATEST

Te6edf3row Tff7b72= Tff7b72selfTff7b72.Td2a8ffproviderTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = Ta5d6ff'Ta5d6ffdatabase_versionTa5d6ff'Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Tffa657intTb4b4b4(Te6edf3rowTb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4]Tb4b4b4)Tb4b4b4, Tff7b72selfTff7b72.Td2a8ffschemaTff7b72.Td2a8ffLATEST_VERSIONTb4b4b4)

Tff7b72if Tff7b72__name__ Tff7b72== Ta5d6ff"Ta5d6ff__main__Ta5d6ff"Tb4b4b4:

Te6edf3unittestTff7b72.Td2a8ffmainTb4b4b4(Tb4b4b4)